/*
 *
 * Copyright (C) 2012 Walter Lütgenau
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * 
 */
package offset.nodes.client.editor.view;

import java.awt.Component;
import java.io.File;

import javax.swing.ImageIcon;
import javax.swing.JTree;
import javax.swing.text.AttributeSet;
import javax.swing.text.Element;
import javax.swing.text.html.HTML;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.DefaultTreeCellRenderer;
import offset.nodes.client.editor.model.EditorModel;
import offset.nodes.client.editor.model.StructureTree;

public class StructureRenderer extends DefaultTreeCellRenderer {

    protected static ImageIcon elementIcon = new ImageIcon(HeadingRenderer.class.getResource("/offset/nodes/client/editor/view/resources/sc_html.png"));
    protected static ImageIcon contentIcon = new ImageIcon(HeadingRenderer.class.getResource("/offset/nodes/client/editor/view/resources/files_text-16.png"));
    public static final String separator = "/";
    EditorModel model;

    public StructureRenderer(EditorModel model) {
        this.model = model;
    }

    public EditorModel getModel() {
        return model;
    }

    public Component getTreeCellRendererComponent(JTree tree, Object value,
            boolean sel,
            boolean expanded,
            boolean leaf, int row,
            boolean hasFocus) {
        Component ret = super.getTreeCellRendererComponent(tree, value, sel, expanded, leaf, row, hasFocus);

        DefaultMutableTreeNode treeNode = (DefaultMutableTreeNode) value;
        if (!(treeNode.getUserObject() instanceof Element))
            return ret;
        
        Element element = (Element) treeNode.getUserObject();
        if (model.isArtificialWithInlineElement(element))
            element = model.getHtmlElement(element).getElement(0);

        if (element.getName().equals(HTML.Tag.CONTENT.toString())) {
            setIcon(contentIcon);
            setText("");
        } else {
            setText(element.getName());
            setIcon(elementIcon);
        }

        return ret;
    }
}
